package com.itheima.utils;

import com.itheima.pojo.Sif;
import org.apache.poi.ss.usermodel.*;

import java.io.InputStream;
import java.time.LocalDateTime;
import java.time.ZoneId;
import java.util.ArrayList;
import java.util.List;

public class SifExcelUtil {

    /**
     * 从Excel文件中读取数据并转换为Sif对象列表。
     *
     * @param inputStream Excel文件的输入流
     * @return Sif对象列表
     * @throws Exception 如果读取文件时发生错误
     */
    public static List<Sif> readExcel(InputStream inputStream) throws Exception {
        if (inputStream == null) {
            throw new IllegalArgumentException("输入流不能为空");
        }
        List<Sif> data = new ArrayList<>();
        try (Workbook workbook = WorkbookFactory.create(inputStream)) {
            Sheet sheet = workbook.getSheetAt(0); // 获取第一个sheet
            // 跳过前1行
            int startRow = 1; // 第二行的索引是1
            System.out.println("当前数据文件的行数:"+sheet.getLastRowNum());
            for (int i = startRow; i <= sheet.getLastRowNum(); i++) {
                System.out.println("停止读取的行数"+i);
                Row row = sheet.getRow(i);
                // 检查时间戳是否为null
                Cell timestampCell = row.getCell(0);
                if (timestampCell == null|| timestampCell.toString().isEmpty())
                {
                    // 如果时间戳为空，则读取文件结束
                    System.out.println("时间戳为空");
                    break;
                }
                LocalDateTime timestamp = getCellValueAsLocalDateTime(timestampCell);
                // 添加日志，查看当前行的单元格内容
                System.out.println("Processing row: " + i);
                for (int j = 0; j < row.getPhysicalNumberOfCells(); j++) {
                    System.out.print("Cell " + j + ": " + row.getCell(j) + " | ");
                }

                Sif sif = new Sif();
                sif.setTimestamp(timestamp);
                sif.setSif(getNumericValueFromCell(row.getCell(2)));
                data.add(sif);
            }
        } catch (Exception e) {
            throw new Exception("读取excel时失败" + e.getMessage(), e);
        }
        return data;
    }

    /**
     * 将单元格值转换为LocalDateTime，如果单元格为空或非日期类型，则返回null。
     *
     * @param cell 单元格对象
     * @return LocalDateTime对象或null
     */
    private static LocalDateTime getCellValueAsLocalDateTime(Cell cell) {
        if (cell == null) {
            return null; // 如果单元格为空，返回null
        }
        if (!DateUtil.isCellDateFormatted(cell)) {
            return null; // 如果单元格不是日期格式，返回null
        }
        return cell.getDateCellValue().toInstant().atZone(ZoneId.systemDefault()).toLocalDateTime();
    }

    /**
     * 将单元格值转换为double，如果单元格为空或非数值类型，则返回null。
     *
     * @param cell 单元格对象
     * @return double值或null
     */
    private static Double getNumericValueFromCell(Cell cell) {
        if (cell == null) {
            System.out.println("单元格为空");
            return null;
        }
        System.out.println("单元格类型: " + cell.getCellType() + ", 单元格值: " + cell.toString());
        switch (cell.getCellType()) {
            case STRING:
                try {
                    // 尝试将字符串转换为Double
                    return Double.parseDouble(cell.getStringCellValue());
                } catch (NumberFormatException e) {
                    // 如果转换失败，打印错误信息并返回null
                    System.out.println("不能将字符串转换为double: " + cell.getStringCellValue());
                    return null;
                }
            case NUMERIC:
                // 如果单元格已经是数值类型，直接获取数值
                return cell.getNumericCellValue();
            default:
                // 对于其他类型，返回null或根据需要处理
                System.out.println("单元格类型不支持转换为double");
                return null;
        }
    }
}